#include "TcpServer.hpp"
#include "CommandExcute.hpp"

#include <memory>

void Usage(std::string proc)
{
    std::cout << "Usage:\n\t" << proc << " local_port\n"
              << std::endl;
}

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        Usage(argv[0]);
        return 1;
    }
    EnableScreen();

    // 端口
    uint16_t port = std::stoi(argv[1]);

    // 白名单
    Command cmd("./safe.txt");

    // 定义服务器
    func_t cmdExec = std::bind(&Command::Excute, &cmd, std::placeholders::_1);
    std::unique_ptr<TcpServer> tcsv = std::make_unique<TcpServer>(port, cmdExec);

    // 启动服务器
    tcsv->InitServer();
    tcsv->Loop();

    return 0;
}